# 1.新增学生
# 2.显示学生
# 3.查找学生
# 4.删除学生
# 0.退出程序
# 存档到文件
# 约定好数据的存储格式~
# 约定把数据保存在和此文件同级目录中,文件名为record.txt
# 文件内容按照行文本的方式来表示~
# 首先这是一个文本文件,里面包含了很多行
# 每一行代表一个学生
# 学生的信息之间使用\t来分割
import os.path
import sys
# 使用这个全局变量来管理所有的学生信息
# 这个列表的每个元素都是一个"字典",每个字典就分别表示了一个学生
students = []


def save():
    """
    用于存档
    :return:
    """
    with open('record.txt', 'w', encoding='utf8') as f:
        for s in students:
            f.write(f"{s['studentId']}\t{s['name']}\t{s['gender']}\t{s['className']}")
        print(f"存档成功!,一共存储了{len(students)}条记录")


def load():
    """
    加载文件内容到students
    :return:
    """
    # 如果存档文件不存在,,贼直接跳过读档流程
    # 为了避免读方式打开文件的时候,文件不存在引起异常
    if not os.path.exists('record.txt'):
        return
    # 读档的时候要保证先把旧的数据给清理干净!
    global students
    students = []
    with open('record.txt', 'r', encoding='utf8') as f:
        for line in f:
            # 针对这一行数据,按照\t进行切分
            # 切分之前,我们要去除末尾的换行
            line = line.strip() # 功能是去掉一个字符串开头和末尾的空白符(空格,换行,回车,制表符...)
            tokens = line.split('\t')
            if len(tokens) != 4:
                print(f"当前行格式存在问题! line = {line}")
                continue
            student = {
                'studentId': tokens[0],
                'name': tokens[1],
                'gender': tokens[2],
                'classNum': tokens[3]
            }
            students.append(student)
    print(f"读档成功! 一共读取了{len(students)}条记录!")

def menu():
    print(' 1. 新增学生')
    print(' 2. 显示学生')
    print(' 3. 查找学生')
    print(' 4. 删除学生')
    print(' 0. 退出程序')
    choice = input(' 请输入您的选择:')
    return choice


def insert():
    print("新增学生开始!")
    studentId = input("请输入学生的学号:")
    name = input("请输入学生的姓名:")
    gender = input("请输入学生的性别:")
    if gender not in ('男','女'):
        print("性别输入的内容不符合要求,新增失败!")
        return
    className = input("请输入学生的班级:")
    # 使用一个字典把上述的信息给聚合起来
    student = {
        'studentId': studentId,
        'name': name,
        'gender': gender,
        'className': className
    }
    global students
    students.append(student)
    # 增加一个保存操作
    save()
    print("新增学生完毕!")


def show():
    # 遍历全局变量的这个列表,把每个学生的信息给打印出来
    print("显示学生开始!")
    for s in students:
        print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
    print(f"显示学生完毕!,一共显示了{len(students)}条数据!")


def find():
    # 根据学生姓名来进行查找
    print("查找学生开始!")
    count = 0
    name = input("请输入要查找的学生姓名:")
    for s in students:
        if name ==s["name"]:
            print(f"[{s['studentId']}]\t{s['name']}\t{s['gender']}\t{s['className']}")
            count += 1
    print(f"查找学生完毕,一共找到了{count}个学生信息!")


def delete():
    print("删除学生开始!")
    count = 0
    studentId = input("请输入要删除的学生学号:")
    # 看看这个学号对应的同学对应的是哪个字典,将这个字典从列表中删除就可以了~
    for s in students:
        if studentId == s["studentId"]:
            print(f"删除 {s['name']} 同学的信息!")
            students.remove(s)
    # 增加一个保存操作
    save()
    print("删除学生结束!")


def main():
    """
    入口函数
    :return:
    """
    # 通过控制台和用户进行交互
    print('-----------------------------------------')
    print('            欢迎来到学生管理系统            ')
    print('-----------------------------------------')
    # 需要在程序启动的时候调用load即可~
    load()
    while True:
        # 通过menu函数来打印出菜单项
        choice = menu()
        if choice == '1':
            # 新增学生
            insert()
        elif choice == '2':
            # 显示所有学生
            show()
        elif choice == '3':
            # 查找学生
            find()
        elif choice == '4':
            # 删除学生
            delete()
        elif choice == '0':
            # 退出程序
            print('goodbye!')
            sys.exit(0)
        else:
            print("您的输入错误,请重新输入!")
            # 进入下次循环,让用户进行重新输入


main()











